import time

from selenium import webdriver


class TestCookielogin():
    def setup_class(self):
        self.driver=webdriver.Chrome()

    def test_get_cookies(self):
        self.driver.get("https://work.weixin.qq.com/wework_admin/loginpage_wx")
        # 等待20s，人工扫码操作
        time.sleep(30)
        #等成功登陆之后，再去获取cookie信息
        cookie=self.driver.get_cookies()
        print(cookie)

    def test_add_cookie(self):
        #访问企业微信主页面
        self.driver.get("https://work.weixin.qq.com/wework_admin/loginpage_wx")
        #读取cookie信息
        cookies=[
            {'domain': '.work.weixin.qq.com', 'httpOnly': False, 'name': 'wxpay.vid', 'path': '/', 'sameSite': 'Lax',
             'secure': False, 'value': '1688855989573989'},
            {'domain': '.work.weixin.qq.com', 'httpOnly': False,
             'name': 'wwrtx.d2st', 'path': '/', 'sameSite': 'Lax',
             'secure': False, 'value': 'a4963922'},
            {
                'domain': '.work.weixin.qq.com', 'httpOnly': True, 'name': 'wwrtx.vst', 'path': '/', 'sameSite': 'Lax',
                'secure': False,
                'value': 'EtYimqOuGKvjILHwQRA3eQ4gC7zuUsSFOZ37vNtR68SFTQ3d9wLfiDung2HW3BbzWDZqlFlY7Z0DSJy8Qlaiv1rlLK7o1uH_A2oUy3kVrNcetV-pzXD5YXDqBkirdf7wzTcyR-wdCyNk4CQ8gDowbYAzp8qF2E53CzZMt2FjfzjtU4sqAUTbk13drLg5hrcRq0ukpWB_IRPJBNRbdyMUrU2cilj2GtgJw6Y9Z4esrfYo-_bX5i6cHVKA3mk2ZnkqHgpRZwsDuDme8r_RY6a32A'},
            {
                'domain': '.work.weixin.qq.com', 'httpOnly': False, 'name': 'wwrtx.cs_ind', 'path': '/',
                'sameSite': 'Lax',
                'secure': False, 'value': ''},
            {'domain': '.work.weixin.qq.com',
             'httpOnly': True, 'name': 'wwrtx.refid',
             'path': '/', 'sameSite': 'Lax', 'secure': False,
             'value': '23475882091826129'},
            {'domain': '.work.weixin.qq.com',
             'expiry': 1756380623, 'httpOnly': False,
             'name': 'wwrtx.c_gdpr', 'path': '/',
             'sameSite': 'Lax', 'secure': False,
             'value': '0'},
            {
                'domain': '.work.weixin.qq.com', 'httpOnly': True, 'name': 'wwrtx.ref', 'path': '/', 'sameSite': 'Lax',
                'secure': False, 'value': 'direct'},
            {'domain': '.work.weixin.qq.com', 'httpOnly': True,
             'name': 'wwrtx.logined', 'path': '/', 'sameSite': 'Lax',
             'secure': False, 'value': 'true'},
            {'domain': '.work.weixin.qq.com',
             'httpOnly': False,
             'name': 'wxpay.corpid',
             'path': '/', 'sameSite': 'Lax',
             'secure': False,
             'value': '1970326783000184'},
            {
                'domain': '.work.weixin.qq.com', 'httpOnly': True, 'name': 'wwrtx.sid', 'path': '/', 'sameSite': 'Lax',
                'secure': False, 'value': '0Dg0XkIGETMzj48vj-SBXp0pEkp8ClMXT08BY_outyBthfjaFF5vaBXQ-IXq0niK'}, {
                'domain': '.work.weixin.qq.com', 'httpOnly': True, 'name': 'wwrtx.ltype', 'path': '/',
                'sameSite': 'Lax',
                'secure': False, 'value': '1'},
            {'domain': '.work.weixin.qq.com', 'expiry': 1727436623, 'httpOnly': False,
             'name': 'ww_lang', 'path': '/', 'sameSite': 'Lax', 'secure': False,
             'value': 'cn,zh'},
            {'domain': '.work.weixin.qq.com', 'httpOnly': False,
             'name': 'wwrtx.vid', 'path': '/', 'sameSite': 'Lax',
             'secure': False, 'value': '1688855989573989'},
            {
                'domain': '.work.weixin.qq.com', 'expiry': 1727436644, 'httpOnly': False, 'name': 'wwrtx.i18n_lan',
                'path': '/', 'sameSite': 'Lax', 'secure': False, 'value': 'zh'}]
        #把cookie信息添加到浏览器
        for cookie in cookies:
            self.driver.add_cookie(cookie)
        #刷新页面
        self.driver.refresh()
        time.sleep(5)